package main.leetcode.clockin.April;

import main.datastruc.ListNode;

/**
 * 23.合并K个链表
 *
 * <p>合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。
 *
 * <p>示例:
 *
 * <p>输入: [   1->4->5,   1->3->4,   2->6 ] 输出: 1->1->2->3->4->4->5->6
 *
 * <p>来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/merge-k-sorted-lists
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class day26 {
    public static void main(String[] args) {
        ListNode head1 = ListNode.creList(new int[] {1, 4, 5});
        ListNode head2 = ListNode.creList(new int[] {1, 3, 4});
        ListNode head3 = ListNode.creList(new int[] {2, 6});
        ListNode[] listNodes = new ListNode[] {head1, head2, head3};
        System.out.println(new day26().mergeKLists(listNodes));
    }

    //    public ListNode mergeKLists(ListNode[] lists) {
    //        int n = lists.length;
    //        if (n < 1) {
    //            return null;
    //        }
    //        if (n < 2) {
    //            return lists[0];
    //        }
    //        PriorityQueue<ListNode> heap = new PriorityQueue<>((Comparator.comparingInt(o ->
    // o.val)));
    //        for (int i = 0; i < n; ++i) {
    //            if (lists[i] != null) {
    //                heap.add(lists[i]);
    //            }
    //        }
    //        ListNode dummy = new ListNode(-1);
    //        ListNode pre = dummy;
    //        while (!heap.isEmpty()) {
    //            pre.next = heap.poll();
    //            pre = pre.next;
    //            if (pre.next != null) {
    //                heap.add(pre.next);
    //            }
    //        }
    //        return dummy.next;
    //    }

    public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0) {
            return null;
        }
        if (lists.length == 1) {
            return lists[0];
        }
        return merge(lists, 0, lists.length - 1);
    }

    private ListNode merge(ListNode[] lists, int l, int r) {
        if (l >= r) {
            return lists[l];
        }
        int mid = l + (r - l >> 1);
        ListNode l1 = merge(lists, l, mid);
        ListNode l2 = merge(lists, mid + 1, r);
        return merge2Lists(l1, l2);
    }

    private ListNode merge2Lists(ListNode l1, ListNode l2) {
        if (l1 == null) {
            return l2;
        }
        if (l2 == null) {
            return l1;
        }
        if (l1.val < l2.val) {
            l1.next = merge2Lists(l1.next, l2);
            return l1;
        } else {
            l2.next = merge2Lists(l1, l2.next);
            return l2;
        }
    }
}
